// cloud_save.js
// 这个文件放在 ComfyUI/web/extensions/cloud_save_workflow/js/ 目录下

// 等待 ComfyUI 前端加载完毕
window.addEventListener("DOMContentLoaded", () => {
    // 检查 app 是否可用
    if (!window.app || !window.app.graph) {
        alert("ComfyUI app 未加载，无法添加云保存功能！");
        return;
    }

    // 创建菜单按钮
    const menu = document.querySelector('.comfy-menu'); // 主菜单
    if (!menu) return;

    // 创建新菜单项
    const cloudSaveBtn = document.createElement('div');
    cloudSaveBtn.className = 'comfy-menu-item';
    cloudSaveBtn.innerText = '云保存工作流';

    // 点击事件
    cloudSaveBtn.onclick = async () => {
        try {
            // 获取当前工作流 JSON
            const workflowJson = window.app.graph.serialize();
            const workflowJsonStr = JSON.stringify(workflowJson, null, 2);

            // 发送到你的后端
            const resp = await fetch('/your_plugin/save_workflow', {
                method: 'POST',
                headers: {'Content-Type': 'application/json'},
                body: workflowJsonStr
            });
            const data = await resp.json();
            if (resp.ok) {
                alert('云保存成功！');
            } else {
                alert('云保存失败：' + (data.error || resp.status));
            }
        } catch (e) {
            alert('云保存异常：' + e.message);
        }
    };

    // 插入到菜单
    menu.appendChild(cloudSaveBtn);
});